package com.weacar.weixin.util;

import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.core.util.QuickWriter;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.io.xml.PrettyPrintWriter;
import com.thoughtworks.xstream.io.xml.XppDriver;
import com.weacar.weixin.controller.SendBonusBean;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import java.io.*;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * User: babyface <xulongfa@weacar.com>
 * Date: 15/10/14
 * Time: 22:50
 */
public class MessageUtil {

    /**
     * 请求消息类型：推送
     */
    public static final String REQ_MESSAGE_TYPE_EVENT = "event";

    /**
     * 事件类型：subscribe(订阅)
     */
    public static final String EVENT_TYPE_SUBSCRIBE = "subscribe";

    /**
     * 事件类型：unsubscribe(取消订阅)
     */
    public static final String EVENT_TYPE_UNSUBSCRIBE = "unsubscribe";

    /**
     * 事件类型：CLICK(自定义菜单点击事件)
     */
    public static final String EVENT_TYPE_CLICK = "CLICK";

    /**
     * 解析微信发来的请求（XML）
     *
     * @param xmlString
     * @return
     * @throws Exception
     */
    @SuppressWarnings("unchecked")
    public static Map<String, String> parseXml(String xmlString) throws Exception {
        // 将解析结果存储在HashMap中
        Map<String, String> map = new HashMap<String, String>();

        // 读取输入流
        SAXReader reader = new SAXReader();
        Document document = reader.read(new ByteArrayInputStream(xmlString.getBytes("utf-8")));
        // 得到xml根元素
        Element root = document.getRootElement();
        // 得到根元素的所有子节点
        List<Element> elementList = root.elements();

        // 遍历所有子节点
        for (Element e : elementList)
            map.put(e.getName(), e.getText());

        return map;
    }

//    public static void main(String[] args) throws Exception {
//        // 将解析结果存储在HashMap中
//        Map<String, String> map = new HashMap<String, String>();
//        String s = "<xml><URL><![CDATA[http://wx.weacar.com/bonus]]></URL><ToUserName><![CDATA[weacar]]></ToUserName><FromUserName><![CDATA[2343]]></FromUserName><CreateTime>1341234123421</CreateTime><MsgType><![CDATA[event]]></MsgType><Event><![CDATA[subscribe]]></Event><Latitude></Latitude><Longitude></Longitude><Precision></Precision><MsgId>1222222312222223122222231222222312222223122222231222222312222223</MsgId></xml>";
//        InputStream inputStream = new ByteArrayInputStream(s.getBytes());
//        // 读取输入流
//        SAXReader reader = new SAXReader();
//        Document document = reader.read(inputStream);
//        // 得到xml根元素
//        Element root = document.getRootElement();
//        // 得到根元素的所有子节点
//        List<Element> elementList = root.elements();
//
//        // 遍历所有子节点
//        for (Element e : elementList)
//            map.put(e.getName(), e.getText());
//
//        // 释放资源
//        inputStream.close();
//
//    }

    /**
     * 文本消息对象转换成xml
     *
     * @param sendBonusBean 文本消息对象
     * @return xml
     */
    public static String messageToXml(SendBonusBean sendBonusBean) {
        xstream.alias("xml", sendBonusBean.getClass());
        return xstream.toXML(sendBonusBean);
    }

    /**
     * 扩展xstream，使其支持CDATA块
     *
     */
    private static XStream xstream = new XStream(new XppDriver() {
        public HierarchicalStreamWriter createWriter(Writer out) {
            return new PrettyPrintWriter(out) {
                // 对所有xml节点的转换都增加CDATA标记
                boolean cdata = true;
                protected void writeText(QuickWriter writer, String text) {
                    if (cdata) {
                        writer.write("<![CDATA[");
                        writer.write(text);
                        writer.write("]]>");
                    } else {
                        writer.write(text);
                    }
                }
            };
        }
    });

}
